package Leetcode;

/**
 * 给定两个大小分别为 m 和 n 的正序（从小到大）数组nums1 和nums2。请你找出并返回这两个正序数组的 中位数 。
 * 算法的时间复杂度应该为 O(log (m+n)) 。
 */
public class 力扣4_寻找两个正序数组的中位数 {
    public static void main(String[] args) {
        System.out.println(findMedianSortedArrays(new int[]{1, 2}, new int[]{3, 4}));
        System.out.println(findMedianSortedArrays(new int[]{0, 0, 0, 0, 0}, new int[]{-1, 0, 0, 0, 0, 0, 1}));
    }

    static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        // 1 合并两个数组
        int idx = 0, idx1 = 0, idx2 = 0;
        int[] nums = new int[nums1.length + nums2.length];
        while (idx1 < nums1.length && idx2 < nums2.length) {
            if (nums1[idx1] < nums2[idx2]) {
                nums[idx++] = nums1[idx1++];
            } else {
                nums[idx++] = nums2[idx2++];
            }
        }
        while (idx1 < nums1.length) {
            nums[idx++] = nums1[idx1++];
        }
        while (idx2 < nums2.length) {
            nums[idx++] = nums2[idx2++];
        }
        // 2 寻找中位数
        int i = nums.length / 2;
        if (nums.length % 2 == 0) {
            return (nums[i] + nums[i - 1]) / 2.0;
        } else {
            return nums[i];
        }
    }

}
